# 配置信息类
from typing import Dict

class TlpConfig(object):
    def __init__(self):
        self.name = 'apps.tlp.tlp_config.TlpConfig'
    
    FRAME_SIZE = 5*60 # 5分钟
    FRAME_STEP = 60 # 隔1分钟取一帧
    WARN_YELLOW = 30*60 # 提前30分钟黄色预警
    WARN_BROWN = 20*60 # 提前20分钟橙色预警
    WARN_RED = 10*60 # 提前10分钟红色预警
    BLOCK_SIZE = 10000 # 第10000条记录存入一个文件
    CLS_NUM = 5 # 共5个类别：0-正常情况；1-黄色预警；2-橙色预警；3-红色报警；4-雷电中；
    SEQ = 0 # 10000条记录文件序号
    device = 'cuda:0'
    writer = None